package com.nzx.general.excel.util;

import com.nzx.common.exception.BizException;
import com.nzx.general.excel.constants.ExcelConstant;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

/**
 * 类描述：Excel校验工具
 *
 * @author nzx
 * @date 2021/11/11 14:36:38
 * */
public class ExcelValidateUtil {

    /**
     * 验证excel格式是否正确
     * 只有 .xls/.xlsx 格式文件符合要求
     * */
    public static boolean checkExcelFileType(MultipartFile file) throws Exception {

        if(file == null){
            throw new BizException("当前Excel文件为空");
        }
        //获取文件名称
        String fileName = file.getOriginalFilename();
        //获取文件名后缀类型
        if(StringUtils.isEmpty(fileName)){
            throw new BizException("当前文件名为空");
        }
        //获取文件后缀类型
        String fileType = fileName.substring(fileName.lastIndexOf("."));
        //判断当前文件是否为Excel要求格式
        return (ExcelConstant.XLS.equals(fileType) || ExcelConstant.XLSX.equals(fileType));
    }
}
